from django.shortcuts import render
from django.views import View
from django.http import JsonResponse
# Create your views here.
# 导入用户模型类进行查询
from apps.users.models import User


class UsernameCountView(View):
    """
        用户名重复判断，查询查询数据库，获取对应的用户数量，如果为0，说明用户不能存在。判断不需要后端实现，只要返回数量即可
    """

    def get(self, request, username):
        # 业务逻辑
        count = User.objects.filter(username=username).count()  # 查询集数据[]

        # 返回json数据
        return JsonResponse({'count': count})


class MobileCountView(View):
    """
        手机号重复判断，查询查询数据库，获取对应的用户数量，如果为0，说明用户不能存在。判断不需要后端实现，只要返回数量即可
    """

    def get(self, request, mobile):
        # 业务逻辑
        count = User.objects.filter(mobile=mobile).count()  # 查询集数据[]

        # 返回json数据
        return JsonResponse({'count': count})
